import graph.graph as gt
import graph.basicproblem as gb

if __name__ == '__main__':
    V = ['A', 'B', 'C', 'D', 'E', 'F']
    E = [['A', 'B', 1], ['A', 'D', 2], ['D', 'F', 3], ['B', 'F', 4], ['B', 'E', 5],
         ['D', 'E', 6], ['D', 'C', 7], ['B', 'C', 8], ['A', 'C', 9], ['E', 'F', 10]]
    [Vp, Ep] = gb.Primspanttree(V, E)
    gt.drawweightgraph(Ep)

    [Vk, Ek] = gb.Kruskalspantree(V, E)
    gt.drawweightgraph(Ek)